package com.lvan.commonutils.response;

import io.swagger.annotations.ApiModelProperty;
import lombok.Data;

import java.util.HashMap;
import java.util.Map;

/**
 * 返回结果公共类
 *
 * @author LuoMeng
 */
@Data
public class Result {

    @ApiModelProperty(value = "是否成功")
    private Boolean success;

    @ApiModelProperty(value = "返回码")
    private Integer code;

    @ApiModelProperty(value = "返回消息")
    private String message;

    @ApiModelProperty(value = "返回数据")
    private Map<String, Object> data = new HashMap<>();

    /**
     * 构造方法私有化，里面的方法都是静态方法，保护属性防篡改
     */
    public Result() {
    }

    /**
     * 成功静态方法
     *
     * @return Result
     */
    public static Result ok() {
        Result result = new Result();
        result.setSuccess(true);
        result.setCode(ResultCode.SUCCESS.getCode());
        result.setMessage(ResultCode.SUCCESS.getMessage());
        return result;
    }

    /**
     * 失败静态方法
     *
     * @return Result
     */
    public static Result error() {
        Result result = new Result();
        result.setSuccess(false);
        result.setCode(ResultCode.ERROR.getCode());
        result.setMessage(ResultCode.ERROR.getMessage());
        return result;
    }

    /**
     * 成功返回状态，带参数
     *
     * @param success 状态
     * @return Result
     */
    public Result success(Boolean success) {
        this.setSuccess(success);
        return this;
    }

    /**
     * 自定义消息返回
     *
     * @param message 消息
     * @return Result
     */
    public Result message(String message) {
        this.setMessage(message);
        return this;
    }

    /**
     * 自定义状态码返回
     *
     * @param code 状态码
     * @return Result
     */
    public Result code(Integer code) {
        this.setCode(code);
        return this;
    }

    /**
     * 数据返回 K/V
     *
     * @param key   key
     * @param value value
     * @return Result
     */
    public Result data(String key, Object value) {
        this.data.put(key, value);
        return this;
    }

    /**
     * 数据返回 Map 对象
     *
     * @param map map Object
     * @return Result
     */
    public Result data(Map<String, Object> map) {
        this.setData(map);
        return this;
    }
}
